package com.itheima.utils;


import javax.sql.DataSource;
import java.sql.Connection;

/**
 * 连接的工具类，它用于从数据源中获取一个连接，并且和线程进行绑定。
 */
public class ConnecctionUtils {
    private ThreadLocal<Connection> t1=new ThreadLocal<Connection>();
    //此DataSource不能new，必须是有spring进行set注入。
    private DataSource dataSource;
    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }
    /**
     * 获取当前线程的连接
     * @return
     */
    public Connection getConnection(){
        //1先从ThreadLocal上获取
        Connection connection=t1.get();
        try {
            //2判断当前线程上是否有连接
            if (connection == null) {
                //2.1从数据源获取连接，并且存入ThreadLocal中。
                connection = dataSource.getConnection();
                //2.1把connection存入threadlocal中
                t1.set(connection);
            }
            //3返回线程上的连接
            return  connection;
        }catch (Exception e){throw new RuntimeException(e);
        }
    }
    /**
     * web应用的涉及问题
     * 把连接和线程解绑
     */
    public void removeConection(){
        t1.remove();
    }
}
